Column stack matrix in a for loop in Python